package com.zhz.leetcode;

import com.zhz.leetcode.entity.Node;

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

/**
 * 589. N 叉树的前序遍历
 * 给定一个 n叉树的根节点 root，返回 其节点值的 前序遍历 。
 * <p>
 * n 叉树 在输入中按层序遍历进行序列化表示，每组子节点由空值 null 分隔（请参见示例）。
 * <p>
 * <p>
 * 示例 1：
 * 输入：root = [1,null,3,2,4,null,5,6]
 * 输出：[1,3,5,6,2,4]
 * <p>
 * 示例 2：
 * 输入：root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
 * 输出：[1,2,3,6,7,11,14,4,8,12,5,9,13,10]
 * <p>
 * <p>
 * 提示：
 * <p>
 * 节点总数在范围[0, 104]内
 * 0 <= Node.val <= 104
 * n 叉树的高度小于或等于 1000
 * <p>
 * <p>
 * 进阶：递归法很简单，你可以使用迭代法完成此题吗?
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/n-ary-tree-preorder-traversal
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class NAryTreePreorderTraversal589 {

    public static void main(String[] args) {

        NAryTreePreorderTraversal589 m = new NAryTreePreorderTraversal589();
        Node node = new Node();
        List<Integer> list = m.preorder(node);
        System.out.println(list);
    }


    /**
     * 使用递归的方式实现
     */
    public List<Integer> preorder(Node root) {
        List<Integer> list = new ArrayList<>();
        if (root == null) {
            return list;
        }
        list.add(root.val);
        List<Node> children = root.children;
        if (children != null && children.size() > 0) {
            for (Node d : root.children) {
                list.addAll(preorder(d));
            }
        }
        return list;
    }


    /**
     * 使用迭代方式
     */
    public List<Integer> preorder1(Node root) {
        List<Integer> list = new ArrayList<>();
        if (root == null) {
            return list;
        }
        Stack<Node> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()) {
            Node d = stack.pop();
            if (d != null) {
                list.add(d.val);
                if (d.children != null) {
                    for (int i = d.children.size() - 1; i >= 0; i--) {
                        stack.push(d.children.get(i));
                    }
                }
            }
        }
        return list;
    }

    /**
     * 基于上面的递归进行优化
     */
    public List<Integer> preorder2(Node root) {
        List<Integer> list = new ArrayList<>();
        test(root,list);
        return list;
    }

    public void test(Node root, List<Integer> list) {
        if (root == null) {
            return ;
        }
        list.add(root.val);
        List<Node> children = root.children;
        if (children != null && children.size() > 0) {
            for (Node d : root.children) {
                test(d,list);
            }
        }
    }


}
